package leetcode._04_字符串;

import org.junit.Test;

import java.util.HashMap;

/**
 * @author pppppp
 * @date 2022/4/12 9:37
 * 给你两个字符串 s1 和 s2 ，写一个函数来判断 s2 是否包含 s1 的排列。如果是，返回 true ；否则，返回 false 。
 * 换句话说，s1 的排列之一是 s2 的 子串 。
 * <p>
 * 示例 1：
 * 输入：s1 = "ab" s2 = "eidbaooo"
 * 输出：true
 * 解释：s2 包含 s1 的排列之一 ("ba").
 * <p>
 * 示例 2：
 * 输入：s1= "ab" s2 = "eidboaoo"
 * 输出：false
 * <p>
 * 提示：
 * 1 <= s1.length, s2.length <= 104
 * s1 和 s2 仅包含小写字母
 */
public class _567_字符串的排列 {
    @Test
    public void T_0() {
        String[][] nums = {{"adc", "dcda"}, {"hello", "ooolleoooleh"}, {"ab", "eidbaooo"}, {"ab", "eidboaoo"}};
        boolean[] ans = {true, false, true, false};
        for (int i = 0; i < nums.length; i++) {
            System.out.println(checkInclusion(nums[i][0], nums[i][1]) == ans[i]);
        }
    }

    /*0.滑动窗口*/
    public boolean checkInclusion(String s1, String s2) {
        if (s1.length() > s2.length()) {
            return false;
        }
        HashMap<Character, Integer> tmap = new HashMap<>();
        HashMap<Character, Integer> curmap = new HashMap<>();
        char[] ch1 = s1.toCharArray();
        int n = s1.length(), count = 0;
        for (char c : ch1) {
            tmap.put(c, tmap.getOrDefault(c, 0) + 1);
        }
        char[] ch2 = s2.toCharArray();
        for (int i = 0; i < n; i++) {
            if (tmap.containsKey(ch2[i])) {
                curmap.put(ch2[i], curmap.getOrDefault(ch2[i], 0) + 1);
                if (curmap.get(ch2[i]) <= tmap.get(ch2[i])) {
                    count++;
                }
            }
        }
        if (count == n) {
            return true;
        }
        int l = 0, r = n;
        while (r < s2.length()) {
            if(tmap.containsKey(ch2[r])){
                curmap.put(ch2[r], curmap.getOrDefault(ch2[r], 0) + 1);
                if (curmap.get(ch2[r]) <= tmap.get(ch2[r])) {
                    count++;
                }
            }

            if(tmap.containsKey(ch2[l])){
                curmap.put(ch2[l], curmap.get(ch2[l]) - 1);
                if (curmap.get(ch2[l]) < tmap.get(ch2[l])) {
                    count--;
                }
            }

            if (count == n) {
                return true;
            }
            l++;
            r++;

        }
        return false;
    }
}
